/*
* Windbg extension for VM traffic manipulation and analysis
* 
* Copyright 2011, isciurus. All rights reserved.
* 
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
* 
* - Redistributions of source code must retain the above copyright notice, this
*   list of conditions and the following disclaimer.
*
* - Redistributions in binary form must reproduce the above copyright notice, this
*   list of conditions and the following disclaimer in the documentation and/or
*   other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
* OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/

#include <map>

using namespace std;


class Mutator
{
public:
	Mutator() { };
	~Mutator();
	HRESULT setMutator(ULONG id, PCHAR scriptFileName, PCHAR filter);
	HRESULT printMutator();

	ULONG _id;
	PCHAR _filter;
	PCHAR _scriptFileName;
};


typedef map<ULONG, Mutator*> mutatorsMap;

class MutationEngine
{
public:

	MutationEngine() { }
	~MutationEngine();

	HRESULT printMutators();
	HRESULT addMutator(PCHAR scriptFileName, PCHAR filter);
	HRESULT removeMutatorById(ULONG mutatorId);

	HRESULT mutationCallback(ULONG64 packetPtr, ULONG packetLength);

private:
	mutatorsMap _mutators;
	static const ULONG _maxMutatorsNum = 100;
};